ai driven treatment